import os


# 根据环境选择数据库配置
TORTOISE_ORM = {
    "connections": {
        "default": {
            "engine": "tortoise.backends.mysql",
            "credentials": {
                "host": os.getenv("MYSQL_HOST", "localhost"),
                "port": int(os.getenv("MYSQL_PORT", 3306)),
                "user": os.getenv("MYSQL_USER", "default"),
                "password": os.getenv("MYSQL_PASSWORD", ""),
                "database": os.getenv("MYSQL_DATABASE", "blog"),
            },
        }
    },
    "apps": {
        "models": {
            "models": [
                "apps.users.models",
                "apps.blog.models",
                "apps.gallery.models",
                "apps.fileserver.models",
                "apps.message.models",
                "apps.system.models",
                "aerich.models",
            ],
            "default_connection": "default",
        }
    },
}

# 其他配置保持不变
DEFAULT_ADMIN_USERNAME = os.getenv("DEFAULT_ADMIN_USERNAME", "admin")
DEFAULT_ADMIN_PASSWORD = os.getenv("DEFAULT_ADMIN_PASSWORD", "admin")
DEFAULT_ADMIN_NICKNAME = os.getenv("DEFAULT_ADMIN_NICKNAME", "管理员")
